package com.gitee.algorithm.linear;

import java.util.Iterator;

/*

线性表的分类：
线性表中数据存储的方式可以是顺序存储，也可以是链式存储，按照数据的存储方式不同，可以把线性表分为顺序
表和链表。
栈是一种基于先进后出(FILO)的数据结构，是一种只能在一端进行插入和删除操作的特殊线性表。
这里采用链表实现栈，也可以使用数组(顺序表的代表)。
 */
public class Stack<T> implements Iterable<T>{
    //记录首节点
    private Node head;

    //栈中的元素个数
    private int N;


    //节点类
    private class Node{
        public T item;
        public Node next;

        public Node(T item, Node next) {
            this.item = item;
            this.next = next;
        }
    }

    public Stack(){
        this.head=new Node(null,null);
        this.N=0;
    }

    //判断当前栈元素个数是否为0
    public boolean isEmpty(){
        return N==0;
    }

    //获取栈中元素的个数
    public int size(){
        return N;
    }

    //把元素压入栈中
    public void push(T t){
        //找到首节点指向的第一个结点
        Node oldFirst = head.next;
        //创建新节点
        Node newNode = new Node(t, null);
        //让首节点指向新节点
        head.next=newNode;
        //让新节点指向原来的第一个结点
        newNode.next=oldFirst;
        //元素个数+1
        N++;
    }

    //把栈中元素弹出
    public T pop(){
        //找到首节点指向的第一个结点
        Node oldFirst = head.next;

        //让首节点指向原来第一个结点的下一个结点
        //直接指向(有问题，万一链表为空，那么oldFirst=null就是空指针)
        //安全性校验
        if (oldFirst==null){
            return null;
        }
        head.next= oldFirst.next;
        //元素个数-1
        N--;
        return oldFirst.item;
    }

    @Override
    public Iterator<T> iterator() {
        return new SIterator();
    }

    private class SIterator implements Iterator{
        private Node n;
        public SIterator(){
            this.n=head;
        }
        @Override
        public boolean hasNext() {
            return n.next!=null;
        }

        @Override
        public Object next() {
            n=n.next;
            return n.item;
        }
    }

}
